package com.example.common.core.utils;

import com.example.common.core.constants.JwtConstants;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;

import java.util.HashMap;
import java.util.Map;

public class JwtUtils {

        /**
     * 生成令牌
     *
     * @param claims 数据
     * @param secret 密钥
     * @return 令牌
     */
    public static String createToken(Map<String, Object> claims, String secret) {
        String token = Jwts.builder()
                .setClaims(claims)
                .signWith(SignatureAlgorithm.HS512,secret)
                .compact();
        return token;
    }

    /**
     * 从令牌中获取数据
     *
     * @param token 令牌
     * @param secret 密钥
     * @return 数据
     */
    public static Claims parseToken(String token, String secret) {
        return Jwts.parser().setSigningKey(secret).parseClaimsJws(token).getBody();
    }

    public static String getUserKey(Claims claims) {
        Object value = claims.get(JwtConstants.LOGIN_USER_KEY);
        return tostr(value);
    }

    public static String getUserId(Claims claims) {
        Object value = claims.get(JwtConstants.LOGIN_USER_ID);
        return tostr(value);
    }

    private static String tostr(Object value) {
        if(value == null) {
            return "";
        }
        return value.toString();
    }

    public static void main(String[] args) {
        Map<String, Object> claims = new HashMap<>();
        claims.put("userId", "1232131232");
        String token = createToken(claims, "AYoxORAwM8qsk+ex4eL2Rasxhi/6s1kf3vE9C9yF7bc=");
        System.out.println(token);
        Claims ret = parseToken(token, "AYoxORAwM8qsk+ex4eL2Rasxhi/6s1kf3vE9C9yF7bc=");
        System.out.println(ret);
    }
}
